import torch

# 检查 GPU 是否可用
print("GPU available:", torch.cuda.is_available())
# 应输出 True

# 检查 cuDNN 是否可用
from torch.backends import cudnn
print(cudnn.is_available())  # 应输出 True

# 查看 CUDA 版本
print("CUDA version:", torch.version.cuda)
# 应与驱动支持的版本一致（如 11.8）

# 查看显卡型号
print("GPU name:", torch.cuda.get_device_name(0))
# 显示显卡型号（如 GeForce MX250）

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("Using device:", device)

print(torch.cuda.get_device_capability())
